# -*- coding: utf-8 -*-


from openai import OpenAI
import os

# 数据库连接配置
# 注意: 实际应用中应该使用更安全的方式存储敏感信息，如环境变量
DB_CONFIG = {
    'host': 'localhost',  # 数据库主机地址
    'port': 3306,          # MySQL默认端口
    'user': 'root',        # 数据库用户名
    'password': '123456'   # 数据库密码
}

# 支持的数据库类型配置
SUPPORTED_DB_TYPES = ['mysql', 'sqlite']  # 支持MySQL和SQLite

# AI服务配置
# 这里使用火山引擎的API服务
AI_CONFIG = {
    'api_key': '',  # API密钥
    'base_url': ''  # API基础URL
}

MODEL_NAME = ''  # 火山方舟推理接入点ID

# 创建OpenAI兼容客户端
client = OpenAI(**AI_CONFIG)

# 可视化配置
VISUALIZATION_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'visualization')
if not os.path.exists(VISUALIZATION_DIR):
    os.makedirs(VISUALIZATION_DIR)